package leetcode;

/*
202. 快乐数
编写一个算法来判断一个数 n 是不是快乐数。
「快乐数」定义为：对于一个正整数，每一次将该数替换为它每个位置上的数字的平方和，然后重复这个过程直到这个数变为 1，也可能是 无限循环 但始终变不到 1。如果 可以变为  1，那么这个数就是快乐数。
如果 n 是快乐数就返回 True ；不是，则返回 False 。
示例：
输入：19
输出：true
解释：
12 + 92 = 82
82 + 22 = 68
62 + 82 = 100
12 + 02 + 02 = 1
*/

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class problems_203 {

    public static void main(String[] args) {
        ListNode node1 = new ListNode(1);
        ListNode node2 = new ListNode(2);
        ListNode node3 = new ListNode(3);
        ListNode node4 = new ListNode(4);
        ListNode node5 = new ListNode(5);
        ListNode node6 = new ListNode(6);
        node1.next = node2;
        node2.next = node3;
        node3.next = node4;
        node4.next = node5;
        node5.next = node6;
        System.out.println(new Solution().removeElements(node1, 2));
    }

    static public class ListNode {
        int val;
        ListNode next;
        ListNode(int x) { val = x; }
    }

    static class Solution {
        public ListNode removeElements(ListNode head, int val) {
            // 追加的虚拟头部，以解决链表头部删除问题
            ListNode virtualHead = new ListNode(0);
            virtualHead.next = head;
            // 遍历用索引
            ListNode index = head, bef = virtualHead;
            while (null != index){
                if(index.val == val){
                    bef.next = index.next; // 切断
                } else {
                    bef = index; // 虚拟头遍历
                }
                index = index.next; // 遍历至下一个
            }
            return virtualHead.next;
        }
    }
}